import MySQLdb

class Equipamento:
    
    def __init__(self,descricao="",marca="",modelo="",acessorio="",hd="",memoria="",processador=""):
        self.idEquipamento = ""
        self.descricao = descricao
        self.marca = marca
        self.modelo = modelo
        self.acessorio = acessorio
        self.hd = hd
        self.memoria = memoria
        self.processador = processador
        self.Diagnostico = ""
        self.TecResponsavel = ""
        
    def getIdEquipamento(self):
        return self.idEquipamento

    def setIdEquipamento(self,novoId):
        self.idEquipamento = novoId


    def getDescricao (self):
        return self.descricao

    def setDescricao (self,novaDescricao):
        self.descricao = novaDescricao



    def getMarca (self):
        return self.marca

    def setMarca (self,novaMarca):
        self.marca = novaMarca



    def getModelo (self):
        return self.modelo

    def setModelo (self,novoModelo):
        self.modelo = novoModelo

        

    def getAcessorio (self):
        return self.acessorio

    def setAcessorio (self,novoAcessorio):
        self.acessorio = novoAcessorio



    def getHd (self):
        return self.hd

    def setHd (self,novoHD):
        self.hd = novoHD



    def getMemoria (self):
        return self.memoria

    def setMemoria (self,novaMemoria):
        self.memoria = novaMemoria



    def getProcessador (self):
        return self.processador

    def setProcessador (self,novoProcessador):
        self.processador = novoProcessador

    def getStatus (self):
        return self.Status

    def setStatus (self,novoStatus):
        self.status = novoStatus

    def getDiagnostico(self):
        return self.Diagnostico

    def setDiagnostico(self,novoDiagnostico):
        self.Diagnostico = novoDiagnostico
        
    def getTecResponsavel(self):
        return self.TecResponsavel

    def setTecResponsavel(self,novoTecResponsavel):
        self.TecResponsavel = novoTecResponsavel

    def insert(self):
        db = MySQLdb.connect("localhost","root","","gtec" )
        cursor = db.cursor()
        
        cursor.execute("INSERT INTO equipamento (descricao,marca,modelo,acessorio,hd,memoria,processador) VALUES ('"+self.descricao+"','"+self.marca+"','"+self.modelo+"','"+self.acessorio+"','"+self.hd+"','"+self.memoria+"','"+self.processador+"')")
        db.commit()
        
    def update(self):
        db = MySQLdb.connect("localhost","root","","gtec" )
        cursor = db.cursor()
        
        cursor.execute("UPDATE equipamento SET descricao = '"+self.descricao+"',marca = '"+self.marca+"',modelo = '"+self.modelo+"',acessorio = '"+self.acessorio+"',hd = '"+self.hd+"',memoria = '"+self.memoria+"',processador = '"+self.processador+"' WHERE idEquipamento = '"+str(self.idEquipamento)+"'")
        db.commit()
        
    def updateStatus(self):
        db = MySQLdb.connect("localhost","root","","gtec" )
        cursor = db.cursor()
        
        cursor.execute("UPDATE equipamento SET status = '"+self.status+"' WHERE idEquipamento = '"+str(self.idEquipamento)+"'")
        db.commit()
        
    def remover(self):
        db = MySQLdb.connect("localhost","root","","gtec" )
        cursor = db.cursor()
        
        cursor.execute("DELETE FROM equipamento WHERE idEquipamento = '"+str(self.idEquipamento)+"'")
        db.commit()
        
    def load(self):
        db = MySQLdb.connect("localhost","root","","gtec" )
        cursor = db.cursor()
                       
        cursor.execute("SELECT * FROM equipamento WHERE idEquipamento =  '"+str(self.idEquipamento)+"'")
        if(cursor.rowcount == 0):
            return False
        else:
             pesq = cursor.fetchall()
             print "\n|ID |"
             for coluna in pesq:
                 print "|",coluna[0],"|" , "\n Descricao: ",coluna[1], "\n Marca: ",coluna[2] , "\n Modelo: ",coluna[3] , "\n Acessorio: ",coluna[4] , "\n HD ", \
            coluna[5] , "\n Memoria: ",coluna[6] , "\n Processador: ",coluna[7] , "\n Status: ",coluna[8],"\n"
        db.commit()

    def insertDiagnostico(self,idEquipamento,Diagnostico,TecResponsavel):
        db = MySQLdb.connect("localhost","root","","gtec" )
        cursor = db.cursor()
        cursor.execute("INSERT INTO diagnostico (IdEquipamento,Diagnostico,TecResponsavel) VALUES (%s,%s,%s)",(idEquipamento,Diagnostico,TecResponsavel))
        db.commit()
